package models

import (
	"context"

	"go.mongodb.org/mongo-driver/mongo"
)

// 定义数据库的数据结构
type Testa struct {
	Name string `json:"name" bson:"name"`
}

// 定义数据库操作方法
type TestaRepository struct {
	collection *mongo.Collection
}

// 构造函数，初始化数据库连接
func NewTestaRepository(db *mongo.Database, collectName string) *TestaRepository {
	return &TestaRepository{collection: db.Collection(collectName)}
}

// 数据库新增数据操作
func (r *TestaRepository) SAdd() error {
	newUser := Testa{Name: "John Doe"}
	_, err := r.collection.InsertOne(context.TODO(), newUser)
	return err
}
